categorical variables using python